Prepare dataset

Start training

TODO